gradle added, test modified to pass