Adding unit tests for the application