Fixed infection spreading. Time to add cell type selectors