Add sobel edge detection