
 - write better tests

