Added a few tests to test the total infection. Fixed a few bugs. Will work on limit infection next. More test cases will come after.
Fixed up comments in a test case. Added limit infection. Will need to do more thorough testing later. Works for general cases currently.