Got SaveBestAfterKEpochs working.