Add extension code