adding wrap function