add more demo code to Main