Add stats dict, remove branch, add annotations