Update aksetup and distribute_setup.py.