Update the code of homework