Changing code to use ForgeDirection instead of Axis

Some code doesn’t work properly yet