Add InfectiousDisease solution code