The code for assignment 1 is basically done now