Fix and add enums