Refactor: better approach for #add (line 3)