simplfied but still dropping 2 in flatten..