little UI fixes for demo