Add fix for visualization bug (related to inserting and removing)