Add typehint for extra safety