fixed namespace bugs
made all visualization functions return a fig, ax tuple