If he's only 5 and these are swim lessons, talk to the swim teacher about it. They can have him do some breathing exercises holding the side of the pool to ensure he's blowing bubbles when his head is down, and making sure he's getting a full breath in. It may be he's taking small shallow breaths and hence not getting enough O2 during the lesson.