fixed import for diagram of diamond problem