Add some tests. Well one