Added second parameter for filename for figure to be saved