Add setup and call correct class