Updated demo to have a line graph