Added more test with generators