Add unit tests to the application