Slight update to math