Assignment errors

Fixing the errors in the assignment