Move main to the top?

Seeing `setUp` and the groups gives a lot of context