Add support for matplotli.Also, put input and output into seperate folders