Add code to output translation for test set
Final fix