update lab
Add tests