fix
finally finish my hw