completed problem set 1 questoin 2