adding capacity for plotting

May remove later