Add test for function map
Add 'for' test