a little tidyup