Added comments to classes and functions