Use .map instead of a for