Add very simple count functionality