clean up rax code