STA641 Assignment 2 Solution 2022