Model improvements for tritium transport in DEMO fuel cycle