change enum to class to hold unit logic.