Adding initial unit tests
Adding helper methods to access stats