Improving the Training of Rectified Flows