lab4 improvement