Changes as per @robdodson's suggestion.

Done!