Update train and test