Add some statment in main to call solution