added save predictions