Add another case for the if statement testing
Add test case for indexing into an array, since the array index can be an expression