Added assertions to test the app