Edited file with assignment 2 code