Added 2 documents to MongoDB
Added unit tests
Some refactoring