Programming Assignment 2

Updated function