Working on test setup