I fixed the problem where all returned images look the same. I started looking into how to parallelize the code and learning photography techniques to make images better. 

I have also made a more presentable diagram, which I will also use in my paper and poster.