adding return check