Add some functionality for getting stats