oc_setreg(i2c, OCI2C_CMD, OCI2C_CMD_STOP);
                        goto out;
                }
-       } else
+       } else {
                msg->buf[i2c->pos++] = oc_getreg(i2c, OCI2C_DATA);
+       }
 
        /* end of msg? */
        if (i2c->pos == msg->len) {
                                i2c->state = STATE_START;
 
                                oc_setreg(i2c, OCI2C_DATA, addr);
-                               oc_setreg(i2c, OCI2C_CMD,  OCI2C_CMD_START);
+                               oc_setreg(i2c, OCI2C_CMD, OCI2C_CMD_START);
                                goto out;
-                       } else
-                               i2c->state = (msg->flags & I2C_M_RD)
-                                       ? STATE_READ : STATE_WRITE;
+                       }
+                       i2c->state = (msg->flags & I2C_M_RD)
+                               ? STATE_READ : STATE_WRITE;
                } else {
                        i2c->state = STATE_DONE;
                        oc_setreg(i2c, OCI2C_CMD, OCI2C_CMD_STOP);
 MODULE_DEVICE_TABLE(of, ocores_i2c_match);
 
 #ifdef CONFIG_OF
-/* Read and write functions for the GRLIB port of the controller. Registers are
+/*
+ * Read and write functions for the GRLIB port of the controller. Registers are
  * 32-bit big endian and the PRELOW and PREHIGH registers are merged into one
- * register. The subsequent registers has their offset decreased accordingly. */
+ * register. The subsequent registers have their offsets decreased accordingly.
+ */
 static u8 oc_getreg_grlib(struct ocores_i2c *i2c, int reg)
 {
        u32 rd;
        int rreg = reg;
+
        if (reg != OCI2C_PRELOW)
                rreg--;
        rd = ioread32be(i2c->base + (rreg << i2c->reg_shift));
 {
        u32 curr, wr;
        int rreg = reg;
+
        if (reg != OCI2C_PRELOW)
                rreg--;
        if (reg == OCI2C_PRELOW || reg == OCI2C_PREHIGH) {
        return 0;
 }
 #else
-#define ocores_i2c_of_probe(pdev,i2c) -ENODEV
+#define ocores_i2c_of_probe(pdev, i2c) -ENODEV
 #endif
 
 static int ocores_i2c_probe(struct platform_device *pdev)
 static int ocores_i2c_remove(struct platform_device *pdev)
 {
        struct ocores_i2c *i2c = platform_get_drvdata(pdev);
+       u8 ctrl = oc_getreg(i2c, OCI2C_CONTROL);
 
        /* disable i2c logic */
-       oc_setreg(i2c, OCI2C_CONTROL, oc_getreg(i2c, OCI2C_CONTROL)
-                 & ~(OCI2C_CTRL_EN|OCI2C_CTRL_IEN));
+       ctrl &= ~(OCI2C_CTRL_EN | OCI2C_CTRL_IEN);
+       oc_setreg(i2c, OCI2C_CONTROL, ctrl);
 
        /* remove adapter & data */
        i2c_del_adapter(&i2c->adap);
        u8 ctrl = oc_getreg(i2c, OCI2C_CONTROL);
 
        /* make sure the device is disabled */
-       oc_setreg(i2c, OCI2C_CONTROL, ctrl & ~(OCI2C_CTRL_EN|OCI2C_CTRL_IEN));
+       ctrl &= ~(OCI2C_CTRL_EN | OCI2C_CTRL_IEN);
+       oc_setreg(i2c, OCI2C_CONTROL, ctrl);
 
        if (!IS_ERR(i2c->clk))
                clk_disable_unprepare(i2c->clk);