minor visual fixes in the demo