Added mock for page class