Improving computational investing hw1 again....