Add docs and remove type hinting