firing rate plotting added