improve lab4