Add start/restart button and logic.