Minor change to got idffernce function