little Changes with trainmode #2