Fixed visualization bug and updated README