Let's set up pytest