Update docs (fix #214)

Update class names in the documentation to fix #214.