First draft of finished problem 2