REF: Using floating point for mean calc, add docstrings