Add comments and descriptions

Add comments and descriptions for classes and functions