I need a work from home policy

A policy generally won't solve the real problem. Ask yourself what's really bothering you?

Is it productivity or communication?

This allows you to then address the real problem with the individuals causing it, without punishing everyone and breaking the trust you have already established.