Fix example
Allow to use klass instead of name

Closes #118