Added a companion object for testing