added gitignore and model training
added model selection