Programming assignment 2 first mod