add some new stuff at lab4