Power Shift control