Improved the speed of training