Changed output in main so that it's simpler for testing