Updated main() to allow multiple levels

Broke everything in the process though