add def main()

added def main() and some temporary code for testing