Couple of fixes in plot3 code