Update InfectionDetection.java

Added stats, cleanup