Remove noise and add docs