problem 3 solved, some stylistic changes to problem 2
Merge pull request #2 from holden-caulfield/problem-3

problem 3 solved, some stylistic changes to problem 2