Adds better typehint