Added restart & level selection