Added if statements: greater than, less than, equal to
Fixed a bug in for loop generation