Adding coding to the Programming Assignment 2