Modelling of charge-exchange induced NBI losses in the COMPASS upgrade tokamak. (8th March 2021)