Also test AugAssign
Add new test cases