For STA 250 homework 2