Add TODO comment to refactor .
Merge pull request #1 from cs-555/comments

Add TODO comment to refactor .