Update extension.rb