Add setup method