Saving plots in custom directory

Plots are now saved in a directory given as an argument.

Simplified function calls

Some unnecessarily complicated parameters have been removed.