Assignment 2: Code with comments

Correct the typo in return statement.