More documentation

Add some description of the classes