Fixed up some style issue