Added a setup function