added __main__ to main file