Wrote most of the code, only the evaluation and model selection bit left, still need to do all the boring writing parts though