added a bit of tests