Fixed data ranges for training

Previously, one of the input features was the actual answer itself.
Definitely not what we need in our input