Solved problem 3 and added tests
Removing pycache file from commit