Moved predicted labels saving