Added code to identify treatment and control districts