Add relevant comment to specify how we fixed the plateau bug.
Fix bug to not show color overlay every time we switch between gradient and input image.