Problem 1 update

Fixed some things on problem 1