First version of the model, working well on the validation set.

initial commit