Fix and test added for image processing