Added missing files for the extension