Add further `ast` node functionality

[zsabbagh]

Add following

• ast.Return
• ast.For
• ast.NamedExpr
• ast.Try