Update solution of homework3