Add the setup method