UI fixes buttons etc, no biggie, just pretty