Added first 2 tests