Add type hinting for public methods