Modify EditShiftCommand Execute method