Move tests next of the code they test