add logger, update main class