fix a bug
finish the second task