Fix documentation

* rename of main class was not reflected in doc