added main()
separated to python files