Cleaned up with() code in plot3