Added missing method implementation