Removed classes from main.

The code is working now that I removed the classes.