(Attempted to) write test code (#3, #21)