Use Berkshelf
Add unit tests