Main class

This is what I wrote.