Submitted code for assingment