Fix plot of selected cells

- Make sure plots of selected cells are accurate.