Bug fixes

Gotta make sure all the buttons work properly, screens are in sync etc.
Don’t make it hard.