ÃLab 4, problem 2 bugfix