Added unit tests for main module - WIP