Add unit tests
Lint the code