I would `return data_map` here, then dedent the stuff in the else.