Add some tests, probably not correct