Style tweaks

Fixes #370