minor changes to make up for visualization stuck, and finishing up on visualization soon