Add tests. Travis to come