Introduce unit tests