Made adding features return Unit and add features to trainDocs