Added a simple visualization