optimize imports, add docs