Added loop for testing