Small updates to GA

Updated the GA code