hw2 update to fix sigma miscalculation