Refactor population_best to take 1 input argument
Update TODO
Minor refactors