Added initial code for the assignment