Fixed stylistic issues from pull request (#31)