Add tests to ensure the app works as intended