Almost have line highlighting working