added flatmap test