Add play count support