Edit comments and minor code cleanup.

Edit test_arbitrary_package_attack.py