Make code beautiful

Little change