improved the demo a little