Initial commit which works with some training and validation set.