removed main and created test