Added tests.
Update