add yaml to imports, add count_units(), update docs