better visualization
fixed bug