Updated comments in some of the class files