Update for more accurate line count