Fixed code to look better

JSHINT