Add type hinting