Added some business logic tests