cleaned up visualization code, including lighter edge colors