Have consise labels
fix for getting infected indices
Return infected and removed indices
Tidy up random detection rates
Fix testing, and add new test for upper bound on detection rates