[#53] Small refactor for clarity

@johnglover Another pull request (same branch), just refactored this code a little for clarity. Does this make it clearer?