added visualisation in 2d