Added more tests for multiple keys scenarios