Adding optional check