The government is consulting on changes as a new social media law goes through its final Parliamentary stages.