Improving the modelling of Gen IV nuclear reactors