Make function style consistent (close #20)

A minor tweak, but I'm like that.