Updated (cleaned up) solution to programming assignment 2 to github repo