Finished my first problem