add demo and make plots more legible