Added better visualization