Add training flag to shifts