fixed import for image for diamond problem