Capability to save the model is added
Bug fixes