Moving shrink into separate function