Taking away a device is not an acceptable approach to discipline in the 1:1 classroom.
Control when the student opens and closes the device. It’s a natural tendency to be bored and pull out your phone from your pocket and play with it. In the same manner, students don’t always intentionally open their computer to play games, just to annoy you.
Lids closed means lids closed. You can take off participation points for not following instructions. You can give them 3 warnings and then notify the parents.
Make your lessons interesting. The student will not be tempted to wander off if your lessons are engaging.
Give students a website or activity that they can work on when there is free time or when they are waiting for others to finish. This a good opportunity to scaffold and give the high achievers an opportunity to be challenged. Tell students where to find these activities in Google Classroom or on your website, so that they always know where to go to find them.