A friend of mine thought the baby might grab his penis if had sex with his wife in her third trimester.

Yes, as long as you're not doing crazy acrobatic sex, having sex with your pregnant partner is 100% fine.