Added restart and styling