Forgot return in function!