Improving homework2.