Working on formatting our main function. Implemented sample polymorphism.