Made Play button able to restart