little Changes with trainmode #3