Trying out mock for testing