Adding note.
Adding testing and saving model.