I must've had a logic error or something when I tried it earlier cause my test was failing.

FIXED