Added setup function to make code concise