Programming Assignment 2 Update

Code updated.