Add logic back in for if else on reply page