SIR Simulation written in Java