switch statement for enum is nicer?