Add stats funtionality