Fixing unit tests and removing run_ensemble()