added support for visualisation in simple GUI