put plotting in visualization file