It's great that plot function returns matplotlib axis!