Add couple of tests