Add code at lifecycle stages