Set up tests

Set up tests correctly