Added  test to main