Assignment2 code submission