Fix Assignment 1 error