Adds initial visualization work