Changed "plane" logic to refer to axis instead and also use the new enum