added couple of tests