Add fitness functon and statistics logging