[FEATURE] Allow return plots