Finally got the first test to pass. Yay!