Added my partial solution to problem 1