[FEATURE] Allow return plots
Merge pull request #5 from lpalomerol/feature/plots

[FEATURE] Allow return plots