fixing color bug and adding 'save to image' button