Add unit tests
Add requirements.txt