generalise handling of input lists, fix axis labels