Add mockito and change census signature to what is expected by the assignment