My changes for the assignment #2

Used the example functions as a frame and added my own code