A few fixes to line classifier