Small change to Demo UI