Added class comments and removed unwanted print statements