Minor changes to line follow code