Updated groovy code