fixing minor details with visualization