Finished Project02 code