Changes to main.
some stuff