tidy up & minor updates