Update rax.py

Small changes