Added code to test the application