Added unit tests for all business logic