Add extension to main