Updating drawing functions for 3D visualisation