Epidemy simulator - week 2 assignement