train solution