Function for Programming Assignment 2