Quick jQuery refactor