Improving plots and adding optional arguments meaning space