Added code for line following (not tested)