* Avoid direct check of class name

So that subclasses are also ok

* Use type hint in function signature.