Improvement and evaluation of the parameterisation of nitrogen oxide production by lightning