Initial tests.
Adding more unit tests