scala. hw2 z1 fixed