make tran sepa as a command arg. change reward model