Add setUp function