Added comments to test class