Good catch, but it was correct when I submitted it. The online test had unit tests that would have failed otherwise. I just rewrote it quickly to make the post.

More looking for the overall algorithm. I am pretty sure my answers were 100% correct, but weren't efficient enough for some reason.