Working fully for black and white images now