Fixed 3d plotting regression, added more labelling