Small edits to please jshint