Updating after-activation method.
Some refactoring on after_activation hook.