Little styling cleanup
Fix #539