Add two more cases
Add two more cases