Improve vim fold support