fixes in lab2