add exit anim and restart functionality