updating how the images data is populated to reflect the change in structure of environment.py