make both fucntions return the axes objects for further plot customisation

[anoopm031]

I have downloaded the code and edited a bit so that the functions will return both the axes (the one where the scatterplot is drawn and the one where the sidebar is drawn) so that users can further customize these plots for more aesthetics.