Add some missing typehints