challenge 04: improve README and tests