Adding jQuery function to establish classes