Clean up old traning code