Added comments to the code made in class