Test01 written more elegantly