Better documenting. Moved imports.