Simple virus simulation now working