Add extension functionality