Modified a few things on demo page to make it look better