:lipstick: Add some tests

Should have had some tests to begin with, but I was in a rush.