move enum into class