The code for the Axon and the Arduino Mega are exactly the same. However the compiler needs to know which you are compiling for, as the .hex file will be different. If you are using Axon code for the Arduino, go into the makefile and change the mcu name from 640 to 1280.