Completed train and test code for LMs