Update to graham's code