Add mock method