MINOR: Cleaned up comments