removed main, adding unit tests