Add 'line count' functionality