Solved problem 1!