started working on tests for Main