Added logic to run test