JS refactoring

Changed jQuery selector for better future functionality