Working on unit testing with Travis CI