Remove __main__, clean ups.