Additional documentation
Fix class name in tests