Added comparison code