Cell phones in class for some reason

So cellphones in classrooms is apparently such a distractive problem, when it could easily be solved if we just let them choose to use their phone or not, teaching the consequence of not having listened in class and having a place for them to put it so it doesn't distract them if they choose to. They will totally still use them if they want , but they have the choice and you can always take it away if they get caught , helping them learn to better hide their phone when employed later in life, like half the planet already does, so that in the end atleast they learn what they wanted to learn.