updated readme

explanation for saving weights is confusing