More work on the plot_expression.py tool in the tools dir