Add extension tests