Added the generation of a colored 2D map along the greyscale image and did some refactoring