Adding assignment one code
Adding assignment 2 code