Printing in main. Data structure not flexible though