Adding comments to web elements and changing the assert condition to when_visible