Add initial test cases