Moved plotting code back to analysis directory