Changed way to add extension