Add more tests for plan API