Update demo train and test