Refactored ARG case of apply_k to use FUN case