Add billing test billing logic