fixed #1. Add main function. 

Add main function. fixed #1.