Fixed saving fig function in marsplot and updated interface to reflect this and commented a little of forward pass