Added the setup stuff in Main. Totally forget it last time.