Write and test F1 score
More model selection