dataroot is an argument
better plotting and saving figures