Update 1: Using Enums