Added some basic visualisation