clean up code - thnks jshint