added ability to plot angular flux