Adding import of Avg back in, but as far as I can tell, not using it?