Improved documentation for all classes