An Improved Data Augmentation Scheme for Model Predictive Control Policy Approximation