Minor tweaks to line follow code