Whitespace
wait_for decorator
Wait for elements to become visible properly
Fix imports
Update test