Change extension import in extensions.py for new flask style