NOTE TO SELF: WRITE MORE TESTS