Add referral tracking