Added a few class comments