Wrap code in jquery block