PART B - PROBLEM 2: IMPLEMENTING A SIMPLE SIMULATION (NO DRUG TREATMENT)