Added test setup

Added tests setup