Add flat map generator