Simplify code

Use List.map2