Now working with real data. Added shuffle and refactored.