Add typehints