add some tests :)