Update rax.py