improve collocation parsing