Added the save back from model and update plot, functions finished bug fixing and documentation starts