Style tweaks
Add doc comment