try and add some tests