small style mod.
bug fix
small change