Added edge check