Worked on visualization stuff