Ditching String comparision by using enum