Add extension method support