Improve section classification