Bug fix. Now loads and trains model on experiment 4