My DS2 has just hit this phase too. At that age they are still learning about what is ok in front of others. Usually they are around 4 when they learn that playing with themselves should be done in private. It is ok to just let him go if you are comfortable with it, or you can gently suggest that he not do it in front of you. Either way, not making a big fuss of it seems to work well.