Update train/test for SLDA