Added some comments to class