cleanup up some UI
add restart button