updated ABM plotting script