added functions to improve visualization
added functions to improve visualization