Added a method to return the processed image