Added new method for visualization