Changes in main function for testing
changes in main function for testing