Added tests for business logic