Testfälle geschrieben, Code verbessert