Code for walking Atlas in simulation