Updating testing framework.

Fixing one of the tests and removing the running classes. IntelliJ has that built in so we don't need it.