Removed a couple of writelines from main method