Improving the prediction of ammonium nitrogen removal through struvite precipitation