Assignment 2 done with comments