Fixing this that propably broke because of ISSUE #101: look into variable and method naming
Fixed class types in docstring