Remove __main__ stuff, not needed