Added comments to classes and methods