Updates on UI and checkmate bugs