Added the project for Problem 4
Finished Problem 4
Cleaned up the code a bit