minor style change and some doc strings