Made main make sense

In my original commit, I wasn't using the functions in main as they were
designed. Also made the TestInput file make more sense =P