Infection model working. tests passing.