Added some documentation to the classes