tidying up problem03