Corrected import to motif_drawer