Implement Programming Assignment 2 functions