Updated to match the example in the readme