Some work on test_ga and cohorts.py.

test_ga working as intending for now. More tests to be added.