add missing type anotations