Adding typehints