Design challenge 1 - update2