added plotting and readme