style fixes plus bug fix