Add unit tests. Test with Travis