make the testing code a bit prettier