Improve code quality slightly

Tested using JShint.