added line following logic
Refactoring of code
Merge pull request #2 from jimmyhlee94/code-refactor

Code refactor