Assignment 4 - Comments changed