Made a stupid mistake when I started the exercise

forgot to add the return for two of the functions