Add typehint