Changed  to take JQuery as parameter in callback

As discussed in #2.