Finished first version of supervised learning (+model selection)