Done with HW4!
have issue in sampling theta from post