Changed aarons code. Made it shuffle each epoch.