added labs4: write tests