Fixed printing out classes
Updated comments