Made example a bit neater