Added the ratio features
Building the trainTest setup, not done yet.