Added required functionality for Programming Assignment 2.