Add unit test working with Travis-CI
Minor fixed