Add a test for with statement
Add With stub implementation