Finished Part 1 of exercise