I figured out a few ways to write the above function. However, I feel like I am both doing it in wrong