scala hw2 fix