Begun working problem 4