I've done this and added a test. Does it look OK?