Added 2 tests to the sample