applied patch
fixed the wrong class docblock
added tests
fixed test