additional tidying in ex. 2