Added a visualisation part
Added some helper methods