- couple of modifications to tidy up the code