Tests are passing, and it kinda works