Changing to after_install