Modified so rho and theta are clear in the visualisation, also fixed mouse location