Got a test working, yay

TODO(Stephen): Actually write tests