Add expected screenshot and remove .only method.