Prettifying demo script a little.