Added check in button and logic to present it