Change IF to Switch via enum
Add quit :)