Check balance in wallet test
Add op= statements
Move code examples away from test directory