update demo to make it looks a bit better