Updated mock with Log functions