Improve visualization formatting