fiixng the visualisation issue 2