Fixes to get the extension to work

Should work now Tyler