Slight modifications to GA algo