minor updates
added 3d plot