Update the test case to add Application class