Forgot to add second function