Added comments for the 3 classes