changed to real images, organized data into separate directory/json file, updated shift functions to handle different dataset sizes