Add some tests ;)