Add initial visualization work