Added a check to to code!