editing demo to run faster

editing demo to run faster