modified assignment1 for submission