Managing nitrogen to restore water quality in China