Add useful __main__ function