Updating the demo to reflect recent design updates.