I think this line needs a return or it'll keep executing the rest of the function.