Add answer for Programming Assignment 2