Forget to add tests.
Added final tests.