Have main take a parameter? Or a system property perhaps?