lab2-3 fix