Add TODO tag for plotting