Modify wording
Add training and testing