Change "complete" plots to use 3d axes for consistency. Refactor plotting code into a helper method.