a bit improvement on visualization