Add test code for if-statements.