Small touches + improved documentation