Add lock_after function