Continued work on visualization