Refactor and add extend functionality via jaxCore.extend()