Fixing visualize function