#7: Add test cases