[ax] Update gem version
[ax] Add more instructions to use the gem