Added the functions for programming assignment 2