Bug fixing of infection function
allow for recovery times to have na's removed, or included if desired
Clean up testing file again