Add functionality for new stat dialog