Test folder will contain the expected output.
Refactor for a better and cleaner logic.