[vim] fix folding of top-level definitions in Python