quick fix on assignment 3 comment, variable names