Re-adding UI for stats