Added more stages test cases