Make it more generic to add any extensions to existing script