fixed small things in demo