Updating extension to be functional