Made changes to solution in problem 2