Changed input parameters

Many functions now accept infected_users, start_time, and end_time
instead.