Start implementing unit tests.

Firs test failing :)