If you hadn't beat me to it, this is exactly what I would have suggested