Added model training and analysis script