Improving computational investment hw1 again.