Cleanup of code.py.
Cleaned up code of arg.py.