Complete code for Programming Assignment #2