fixed main to actually call class functions