Add restartLevel() function