Cleaned up some code and added more info to the infection command.