add comment
simplified using map, passes final tests