update the extension template to new class system