Add get_ages function
predict.py main: refactor training data import
Refactor importing data for prediction