Added Dustin's code to main.