added restart and check buttons with functionality