Add readme, improve comments, remove model selection