UI fixes for 4.1.0