train part seem to work