Fixed main class to support new logging method.