Added code tests + images for large data set