Add data model and main function for testing purpose