Added restart button & repositioned