Update UI, fix some display bugs, add restart feature