Add type hinting

@georgehrke