Add in afterrender hook