Add AX submodule