Adding code plus mock