Adding plot3 code and figure