Add additional test cases