Fixed some memory useage, added colormaps