Added __main__ function