Corrected __main__ part.