Great work! This code looks better already.
We can remove even more, however. There's one return statement in this code that we can change from explicit to implicit!
Recall that Ruby will automatically return the value of the last expression it evaluates.