added it to restart and level