adding in assignment 1 cs4540