Added fitness and changed generate to work with it