Added tests to ensure working of the Generator